import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.ValueSource;

/**
 * 爬楼梯
 * 题目：假设你正在爬楼梯。需要 n阶你才能到达楼顶。
 * 每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢？
 * 注意：给定 n 是一个正整数。
 * <p>
 * 示例 1：
 * 输入： 2
 * 输出： 2
 * 解释： 有两种方法可以爬到楼顶。
 * 1.  1 阶 + 1 阶
 * 2.  2 阶
 * <p>
 * 示例 2：
 * 输入： 3
 * 输出： 3
 * 解释： 有三种方法可以爬到楼顶。
 * 1.  1 阶 + 1 阶 + 1 阶
 * 2.  1 阶 + 2 阶
 * 3.  2 阶 + 1 阶
 * <p>
 * 来源：力扣（LeetCode-70）
 * 链接：https://leetcode-cn.com/problems/climbing-stairs
 *
 * @author godfrey
 * @since 2020-11-11
 */
public class ClimbStairs extends BaseTest {

    @DisplayName("斐波拉契-时间复杂度O(n)，空间复杂度O(1)")
    @ParameterizedTest
    @ValueSource(ints = {2, 3, 4, 5, 6})
    void climbStairs(int n) {
        int pre = 1, rear = 2;
        int sum = n;
        for (int i = 3; i <= n; ++i) {
            sum = pre + rear;
            pre = rear;
            rear = sum;
        }
        System.out.println(sum);
    }

    @DisplayName("斐波那契数列的公式-时间复杂度O(logn)，空间复杂度O(1)")
    @ParameterizedTest
    @ValueSource(ints = {2, 3, 4, 5, 6})
    void climbStairs2(int n) {
        final double s = Math.sqrt(5);
        System.out.println(((int) Math.floor((Math.pow((1 + s) / 2, n + 1) + Math.pow((1 - s) / 2, n + 1)) / s + 0.5)));
    }

    @DisplayName("动态规划-时间复杂度O(n)，空间复杂度O(1)")
    @ParameterizedTest
    @ValueSource(ints = {2, 3, 4, 5, 6})
    void climbStairs3(int n) {
        if (n <= 2) {
            System.out.println(n);
            return;
        }
        //初始化
        int[] dp = new int[n + 1];
        //边界
        dp[1] = 1;
        dp[2] = 2;
        for (int i = 3; i < dp.length; ++i) {
            //转移方程
            dp[i] = dp[i - 1] + dp[i - 2];
        }
        System.out.println(dp[n]);
    }
}
